#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
                            _ooOoo_  
                           o8888888o  
                           88" . "88  
                          (|  -_-  |)  
                           O\  =  /O  
                        ____/`---'\____  
                      .   ' \\| |// `.  
                       / \\||| : |||// \  
                     / _||||| -:- |||||- \  
                       | | \\\ - /// | |  
                     | \_| ''\---/'' | |  
                      \ .-\__ `-` ___/-. /  
                   ___`. .' /--.--\ `. . __  
                ."" '< `.___\_<|>_/___.' >'"".  
               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
                 \ \ `-. \_ __\ /__ _/ .-` / /  
         ======`-.____`-.___\_____/___.-`____.-'======  
                            `=---='  
  
         .............................................  
                  佛祖镇楼                  BUG辟易  
          佛曰:  
                  写字楼里写字间，写字间里程序员；  
                  程序人员写程序，又拿程序换酒钱。  
                  酒醒只在网上坐，酒醉还来网下眠；  
                  酒醉酒醒日复日，网上网下年复年。  
                  但愿老死电脑间，不愿鞠躬老板前；  
                  奔驰宝马贵者趣，公交自行程序员。  
                  别人笑我忒疯癫，我笑自己命太贱；  
                  不见满街漂亮妹，哪个归得程序员？ 
'''
# @File  : modbus_slave_1.py
# @Author: huguangzhi
# @Drivce: Thinkpad E470
# @ContactEmail : huguangzhi@ucsdigital.com.com 
# @ContactPhone : 13121961510 
# @Date  : 2021/7/16 - 17:14
# @Desc  :


import sys
import time

import modbus_tk
import modbus_tk.defines as cst
from modbus_tk import modbus_tcp


class modbus_touch():
    def __init__(self, ip="localhost", port=60002):
        self.logger = modbus_tk.utils.create_logger(name="console", record_format="%(message)s")
        self.server = modbus_tcp.TcpServer(port=port, address=ip)
        self.logger.info("create TCP ip:{} port:{}".format("localhost", str(60002)))
        self.server.start()

        # TCP server添加一个slave
        slave_1 = self.server.add_slave(1)
        # 添加块地址，对应的分别为块名字，快类型，块开始地址，块的容量大小
        slave_1.add_block('0', cst.HOLDING_REGISTERS, 40011, 100)

    def add_slave(self, slave_id: int):
        self.server.add_slave(slave_id)
        # 输入告诉已经添加了slave信息
        self.logger.info('done: slave %d added\r\n' % slave_id)

    def add_block(self, slave_id: int, name: str, block_type=cst.HOLDING_REGISTERS, address=0, size=10):
        slave = self.server.get_slave(slave_id)
        slave.add_block(name, block_type, address, size)
        self.logger.info('done: block %s added\r\n' % name)

    def set_value(self, slave_id: int, name: str,address:int, values:list):
        slave = self.server.get_slave(slave_id)
        # 主要是这里设置slave的值 其中value 是一个数组的形式
        slave.set_values(name, address, values)
        values = slave.get_values(name, address, len(values))
        # 输出返回了写入的信息
        self.logger.info('done: values written: %s\r\n' % str(values))

    def get_values(self, slave_id: int, name: str,address:int, length:int):
        slave = self.server.get_slave(slave_id)
        values = slave.get_values(name, address, length)
        self.logger.info('done: values read: %s\r\n' % str(values))




if __name__ == "__main__":
    modbus = modbus_touch()
    modbus.add_slave(2)
    modbus.add_block(2, 2)
    while True:
        print(".")
        time.sleep(1000)
        modbus.set_value(2, 2, 0, [2, 3, 4])
        print(".")
        print(modbus.get_values(2, 2, 1, 10))

    '''
    1.set_values 1 0 1 2 3 6 12解释

    第一个参数1是从站的号，第二个参数0是从站的块序号，第三个参数1是开始的地址，

    后面的参数2 3 6 12是从地址1开始写，有几个参数写几个地址，依次后排

    2.get_values 1 0 1 4解释：

    第一个参数1是从站的号，第二个参数0是从站的块序号，第三个参数1是开始的地址，

    后面的参数4是从开始地址往后数4个进行读取。

    可以连接一个实际的从站，也可以连接一个调试助手的从站，运行结果如下：

                '''
